from shutil import which

import matplotlib.pyplot as plt

# 15-1 立方：数字的三次方被称为其立方。请绘制一个图形，显示前5个整数的立方值，再绘制一个图形，显示前5000个整数的立方值。
# 15-2 彩色立方：给你前面绘制的立方图指定颜色映射。

# int_values = list(range(1, 6))
# y_values = [x ** 3 for x in int_values]

int_values = list(range(1, 5001))
y_values = [x ** 3 for x in int_values]
# plt.plot(int_values, y_values, linewidth=3)
plt.scatter(int_values, y_values, c=y_values,  # 颜色根据 y 值变化
            cmap='Blues',
            edgecolors='none',
            s=4,
            vmin=0, vmax=max(y_values)
            )

plt.title("Cubes of Integers", fontsize=20)
plt.xlabel("Value", fontsize=14)
plt.ylabel("Cubes Value", fontsize=14)
plt.tick_params(axis='both', which='major', labelsize=12)
plt.show()
